Continue a recursive function in python